Why am I getting a NameError when I pass an argument to a function in python